/*
 * (C) Copyright 2013-2015 Freescale Semiconductor, Inc.
 *
 * SPDX-License-Identifier:	GPL-2.0+
 */

#include <common.h>
#include <asm/io.h>
#include <div64.h>
#include <asm/arch/imx-regs.h>
#include <asm/arch/clock.h>

static struct pit_reg *cur_pit = (struct pit_reg *)PIT_BASE_ADDR;

DECLARE_GLOBAL_DATA_PTR;

#define TIMER_LOAD_VAL    0xffffffff

static inline unsigned long long tick_to_time(unsigned long long tick)
{
    tick *= CONFIG_SYS_HZ;
    do_div(tick, mxc_get_clock(MXC_PERIPHERALS_CLK));

    return tick;
}

static inline unsigned long long us_to_tick(unsigned long long usec)
{
    usec = usec * mxc_get_clock(MXC_PERIPHERALS_CLK)  + 999999;
    do_div(usec, 1000000);

    return usec;
}

int timer_init(void)
{
    __raw_writel(0, &cur_pit->mcr);

    __raw_writel(TIMER_LOAD_VAL, &cur_pit->ldval1);
    __raw_writel(0, &cur_pit->tctrl1);
    __raw_writel(1, &cur_pit->tctrl1);

    gd->arch.tbl = 0;
    gd->arch.tbu = 0;

    return 0;
}

unsigned long long get_ticks(void)
{
    ulong now = TIMER_LOAD_VAL - __raw_readl(&cur_pit->cval1);

    /* increment tbu if tbl has rolled over */
    if (now < gd->arch.tbl)
        gd->arch.tbu++;
    gd->arch.tbl = now;

    return (((unsigned long long)gd->arch.tbu) << 32) | gd->arch.tbl;
}

ulong get_timer_masked(void)
{
    return tick_to_time(get_ticks());
}

ulong get_timer(ulong base)
{
    return get_timer_masked() - base;
}

/* delay x useconds AND preserve advance timstamp value */
void __udelay(unsigned long usec)
{
    unsigned long long start;
    ulong tmo;

    start = get_ticks();            /* get current timestamp */
    tmo = us_to_tick(usec);            /* convert usecs to ticks */
    while ((get_ticks() - start) < tmo)
        ;                /* loop till time has passed */
}

/*
 * This function is derived from PowerPC code (timebase clock frequency).
 * On ARM it returns the number of timer ticks per second.
 */
ulong get_tbclk(void)
{
    return mxc_get_clock(MXC_PERIPHERALS_CLK);
}
